package cn.xinfei.xdecision.common.utils.util;

import org.springframework.aop.framework.AdvisedSupport;
import org.springframework.aop.support.AopUtils;
import org.springframework.beans.BeansException;
import org.springframework.beans.factory.NoSuchBeanDefinitionException;
import org.springframework.beans.factory.config.BeanFactoryPostProcessor;
import org.springframework.beans.factory.config.ConfigurableListableBeanFactory;
import org.springframework.stereotype.Component;

import java.lang.annotation.Annotation;
import java.lang.reflect.Field;
import java.util.Map;


@Component
public class SpringContextUtil implements BeanFactoryPostProcessor {

	private static ConfigurableListableBeanFactory beanFactory; // Spring应用上下文环境

	@Override
	public void postProcessBeanFactory(ConfigurableListableBeanFactory beanFactory) throws BeansException {
		SpringContextUtil.beanFactory = beanFactory;
	}

	public static ConfigurableListableBeanFactory getBeanFactory() {
		return beanFactory;
	}

	/**
	 * 获取对象
	 *
	 * @param name
	 * @return Object 一个以所给名字注册的bean的实例
	 * @throws BeansException
	 *
	 */
	@SuppressWarnings("unchecked")
	public static <T> T getBean(String name) throws BeansException {
		return (T) beanFactory.getBean(name);
	}

	public static <T> Map<String, T> getBeansOfType(Class<T> requiredType) {
		return beanFactory.getBeansOfType(requiredType);
	}

	/**
	 * 获取所有带有特定注解的bean
	 *
	 * @param annotationType 注解类型
	 * @param <T> bean类型
	 * @return bean名称-bean映射
	 */
	public static <T> Map<String, T> getBeansWithAnnotation(Class<? extends Annotation> annotationType) {
		return (Map<String, T>) beanFactory.getBeansWithAnnotation(annotationType);
	}

	/**
	 * 获取类型为requiredType的对象
	 *
	 * @param clz
	 * @return
	 * @throws BeansException
	 *
	 */
	public static <T> T getBean(Class<T> clz) throws BeansException {
		@SuppressWarnings("unchecked")
		T result = (T) beanFactory.getBean(clz);
		return result;
	}

	/**
	 * 如果BeanFactory包含一个与所给名称匹配的bean定义，则返回true
	 *
	 * @param name
	 * @return boolean
	 */
	public static boolean containsBean(String name) {
		return beanFactory.containsBean(name);
	}

	/**
	 * 判断以给定名字注册的bean定义是一个singleton还是一个prototype。 如果与给定名字相应的bean定义没有被找到，将会抛出一个异常（NoSuchBeanDefinitionException）
	 *
	 * @param name
	 * @return boolean
	 * @throws NoSuchBeanDefinitionException
	 *
	 */
	public static boolean isSingleton(String name) throws NoSuchBeanDefinitionException {
		return beanFactory.isSingleton(name);
	}

	/**
	 * @param name
	 * @return Class 注册对象的类型
	 * @throws NoSuchBeanDefinitionException
	 *
	 */
	public static Class<?> getType(String name) throws NoSuchBeanDefinitionException {
		return beanFactory.getType(name);
	}

	/**
	 * 如果给定的bean名字在bean定义中有别名，则返回这些别名
	 *
	 * @param name
	 * @return
	 * @throws NoSuchBeanDefinitionException
	 *
	 */
	public static String[] getAliases(String name) throws NoSuchBeanDefinitionException {
		return beanFactory.getAliases(name);
	}

	public static <T> T getTarget(Class<T> clz) {
		T beanInstance=getBean(clz);
		if (!AopUtils.isAopProxy(beanInstance)) {
			return beanInstance;
		} else if (AopUtils.isCglibProxy(beanInstance)) {
			try {
				Field h = beanInstance.getClass().getDeclaredField("CGLIB$CALLBACK_0");
				h.setAccessible(true);
				Object dynamicAdvisedInterceptor = h.get(beanInstance);
				Field advised = dynamicAdvisedInterceptor.getClass().getDeclaredField("advised");
				advised.setAccessible(true);
				Object target = ((AdvisedSupport) advised.get(dynamicAdvisedInterceptor)).getTargetSource().getTarget();
				return (T)target;
			} catch (Exception e) {
				e.printStackTrace();
			}

		}
		return null;

	}


}
